import pymongo
from pymongo.collection import Collection

# 连接 MongoDB 数据库
client = pymongo.MongoClient(host='192.168.2.162', port=27017)
db = client['demo_data']

def handle_save_data(item):
    """
    根据 item 类型将数据保存到相应的 MongoDB 集合中。
    支持的类型包括：douyin_item、kuaishou_item 和 jinritoutiao_item。
    """
    try:
        # 抖音
        if item['item_type'] == 'douyin_item':
            print('抖音')
            douyin_data_collection = Collection(db, 'douyin')
            douyin_data_collection.insert_one(item)
        # 快手
        elif item['item_type'] == 'kuaishou_item':
            print('快手')
            kuaishou_data_collection = Collection(db, 'kuaishou')
            kuaishou_data_collection.insert_one(item)
        # 今日头条
        elif item['item_type'] == 'jinritoutiao_item':
            jinritoutiao_data_collection = Collection(db, 'jinritoutiao')
            jinritoutiao_data_collection.insert_one(item)
        else:
            print("未知类型：", item['item_type'])
    except pymongo.errors.PyMongoError as e:
        print(f"Error saving data: {e}")
